package top.codecrab.eblog.utils;

import cn.hutool.core.util.StrUtil;
import cn.smallbun.screw.core.Configuration;
import cn.smallbun.screw.core.engine.EngineConfig;
import cn.smallbun.screw.core.engine.EngineFileType;
import cn.smallbun.screw.core.engine.EngineTemplateType;
import cn.smallbun.screw.core.execute.DocumentationExecute;
import cn.smallbun.screw.core.process.ProcessConfig;
import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceWrapper;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;

import javax.sql.DataSource;
import java.sql.Driver;
import java.util.ArrayList;
import java.util.List;

/**
 * @author 王刚
 * @since 2022年08月24日 11:17
 */
public class DbDocUtil {

    /**
     * 文档生成
     */
    public static void documentGeneration(Class<? extends Driver> driverClass, String jdbcUrl, String username, String password, String docPath, EngineFileType fileType, boolean hasRemark) {
        //数据源
        HikariConfig hikariConfig = new HikariConfig();
        hikariConfig.setDriverClassName(driverClass.getName());
        hikariConfig.setJdbcUrl(jdbcUrl);
        hikariConfig.setUsername(username);
        hikariConfig.setPassword(password);
        //设置可以获取tables remarks信息
        hikariConfig.addDataSourceProperty("useInformationSchema", Boolean.toString(hasRemark));
        hikariConfig.setMinimumIdle(2);
        hikariConfig.setMaximumPoolSize(5);
        DataSource dataSource = new HikariDataSource(hikariConfig);
        documentGeneration(docPath, fileType, dataSource);
    }

    /**
     * 生成数据库设计文档
     *
     * @param docPath                生成的路径，包含文件名，最后一个"/"后面的为文件名，不管后缀
     * @param fileType               数据库设计文档的类型
     * @param druidDataSourceWrapper 数据源信息
     */
    public static void documentGeneration(String docPath, EngineFileType fileType, DruidDataSourceWrapper druidDataSourceWrapper) {
        Driver driver = druidDataSourceWrapper.getDriver();
        String url = druidDataSourceWrapper.getUrl();
        String username = druidDataSourceWrapper.getUsername();
        String password = druidDataSourceWrapper.getPassword();
        documentGeneration(driver.getClass(), url, username, password, docPath, fileType, true);
    }

    /**
     * 生成数据库设计文档
     *
     * @param docPath    生成的路径，包含文件名，最后一个"/"后面的为文件名，不管后缀
     * @param fileType   数据库设计文档的类型
     * @param dataSource 数据源
     */
    public static void documentGeneration(String docPath, EngineFileType fileType, DataSource dataSource) {
        if (StrUtil.isBlank(docPath)) {
            return;
        }

        String[] fileNameType = StrUtil.subAfter(docPath, "/", true).split("\\.");
        String filePath = StrUtil.subBefore(docPath, "/", true);
        String fileName = fileNameType[0];

        //生成配置
        EngineConfig engineConfig = EngineConfig.builder()
            //生成文件路径
            .fileOutputDir(filePath)
            //打开目录
            .openOutputDir(true)
            //文件类型
            .fileType(fileType)
            //生成模板实现
            .produceType(EngineTemplateType.freemarker)
            //自定义文件名称
            .fileName(fileName)
            .build();

        //忽略表
        List<String> ignoreTableName = new ArrayList<>();

        //忽略表前缀
        List<String> ignorePrefix = new ArrayList<>();

        //忽略表后缀    
        List<String> ignoreSuffix = new ArrayList<>();

        ProcessConfig processConfig = ProcessConfig.builder()
            //指定生成逻辑、当存在指定表、指定表前缀、指定表后缀时，将生成指定表，其余表不生成、并跳过忽略表配置	
            //根据名称指定表生成
            .designatedTableName(new ArrayList<>())
            //根据表前缀生成
            .designatedTablePrefix(new ArrayList<>())
            //根据表后缀生成	
            .designatedTableSuffix(new ArrayList<>())
            //忽略表名
            .ignoreTableName(ignoreTableName)
            //忽略表前缀
            .ignoreTablePrefix(ignorePrefix)
            //忽略表后缀
            .ignoreTableSuffix(ignoreSuffix)
            .build();

        //配置
        Configuration config = Configuration.builder()
            //版本
            .version("1.0.0")
            //描述
            .description("数据库设计文档生成")
            //数据源
            .dataSource(dataSource)
            //生成配置
            .engineConfig(engineConfig)
            //生成配置
            .produceConfig(processConfig)
            .build();

        //执行生成
        new DocumentationExecute(config).execute();
    }

}
